Training set

Log probability vs time

Training set predictions

Test set predictions